####################################
# CER-bandwidth
####################################

rm(list=ls())

#sink("~/Dropbox/Crime Chile/11_replication/09_rdrobust_CER_crime.txt")

library(Hmisc)
library(ggplot2)
library(stargazer)
library(foreign)
library(rdrobust) 
library(rdd)
library(readstata13)

################
# Prepare data 
################

# read data
d = read.dta13("~/Dropbox/Crime Chile/11_replication/local_crime_data_chile_2022january.dta")   
names(d)

############################
# RDplots
############################

# Homicides
summary(rdrobust(d$homicide_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA5a.pdf", 
          width=7, 
          height=7)
rdplot(y = d$homicide_rate_s, x = d$margin, h= 0.115, nbins = 100, subset = -0.115 <= d$margin & d$margin <= 0.115,
       binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-0.549 , 0.192]",
       y.label = "Homicide", x.label = "Margin of victory")
dev.off() 

# Rapes
summary(rdrobust(d$rape_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA5b.pdf", 
          width=7, 
          height=7)
(rdplot(y = d$rape_rate_s, x = d$margin,  h= 0.111, nbins = 100, subset = -0.111 <= d$margin & d$margin <= 0.111, 
        binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-0.352 , 0.098]",
        y.label = "Rape", x.label = "Margin of victory"))
dev.off()

# Assault
summary(rdrobust(d$assault_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA5c.pdf", 
          width=7, 
          height=7)
(rdplot(y = d$assault_rate_s, x = d$margin,  h= 0.142, nbins = 100, subset = -0.142 <= d$margin & d$margin <= 0.142, 
        binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-0.390 , 0.113]",
        y.label = "Assault", x.label = "Margin of victory"))
dev.off()

# Theft
summary(rdrobust(d$theft_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA6a.pdf", 
          width=7, 
          height=7)
(rdplot(y = d$theft_rate_s, x = d$margin,  h= 0.069, nbins = 100, subset = -0.069 <= d$margin & d$margin <= 0.069, 
        binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-0.961 , -0.232]",
        y.label = "Theft", x.label = "Margin of victory"))
dev.off()

# Robbery
summary(rdrobust(d$robbery_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA6b.pdf", 
          width=7, 
          height=7)
(rdplot(y = d$robbery_rate_s, x = d$margin,  h= 0.076, nbins = 100, subset = -0.076 <= d$margin & d$margin <= 0.076, 
        binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-1.012 , -0.114]",
        y.label = "Robbery", x.label = "Margin of victory"))
dev.off()

# Robbery by surprise
summary(rdrobust(d$robbery_surprise_rate_s,d$margin,cluster = d$cluster,all=TRUE, bwselect = "cerrd"))
cairo_pdf(file="~/Dropbox/Crime Chile/11_replication/figureA6c.pdf", 
          width=7, 
          height=7)
(rdplot(y = d$robbery_surprise_rate_s, x = d$margin,  h= 0.079, nbins = 100, subset = -0.079 <= d$margin & d$margin <= 0.079, 
        binselect="esmv", kernel="triangular", col.lines = "black", col.dots = "black", p=1, y.lim = c(-1, 1), title = "                                       Robust CI: [-1.036 , -0.111]",
        y.label = "Robbery by surprise", x.label = "Margin of victory"))
dev.off()

#######################
# Tables
#######################

# Homicide
summary(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw1 = as.numeric(rdrobust(d$homicide_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

# Rape
summary(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw2 = as.numeric(rdrobust(d$rape_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

# Assault 
summary(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw3 = as.numeric(rdrobust(d$assault_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

# Theft
summary(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw4 = as.numeric(rdrobust(d$theft_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

# Robbery
summary(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw5 = as.numeric(rdrobust(d$robbery_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

# Robbery by surprise
summary(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd"))

pe6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$Estimate[1])
pv6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$pv[3])
ci6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$ci[3,])
oss6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[1] + rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N[2])
ess6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[1] + rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$N_h[2])
bw6 = as.numeric(rdrobust(d$robbery_surprise_rate_s,d$margin,all=TRUE,cluster = d$cluster, bwselect = "cerrd")$bws[1,1])

pe = c(pe1,pe2,pe3,pe4,pe5,pe6)
pv = c(pv1,pv2,pv3,pv4,pv5,pv6)
ci_lower = c(ci1[1],ci2[1],ci3[1],ci4[1],ci5[1],ci6[1])
ci_upper = c(ci1[2],ci2[2],ci3[2],ci4[2],ci5[2],ci6[2])
oss = c(oss1,oss2,oss3,oss4,oss5,oss6)
ess = c(ess1,ess2,ess3,ess4,ess5,ess6)
bw = c(bw1,bw2,bw3,bw4,bw5,bw6)
variable = c("Homicide","Rape","Assault","Theft","Robbery","Robbery by surprise")
data = data.frame(variable,pe,pv,ci_lower,ci_upper,oss,ess,bw)

colnames(data) = c("","Point Estimate","Robust P-value","Robust 95% Confidence Interval lower bound","Robust 95% Confidence Interval upper bound","Overall sample size","Effective sample size","MSE bandwidth")
data

stargazer(data, summary=FALSE, rownames=FALSE, out="~/Dropbox/Crime Chile/11_replication/tableA16.html")

sink()
